package org.example.controller;

import org.example.service.DataService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

/**
 * 数据生成控制器
 */
@RestController
@RequestMapping("/api/data-generator")
public class DataGeneratorController {
    
    @Autowired
    private DataService dataService;
    
    /**
     * 生成并插入数据
     */
    @PostMapping("/generate")
    public String generateData() {
        try {
            dataService.generateAndInsertData();
            return "数据生成成功！";
        } catch (Exception e) {
            e.printStackTrace();
            return "数据生成失败：" + e.getMessage();
        }
    }
    
    /**
     * 清空现有数据
     */
    @DeleteMapping("/clear")
    public String clearData() {
        try {
            // 这里可以添加清空数据的方法
            return "数据清空成功！";
        } catch (Exception e) {
            e.printStackTrace();
            return "数据清空失败：" + e.getMessage();
        }
    }
}
